import torch

# 模型保存, 一般后缀为 .pth
torch.save(model.state_dict(), 'path/filename.pth')
